from sentence_transformers import SentenceTransformer
import numpy as np

class TextProcessor:
    def __init__(self):
        self.model = SentenceTransformer('BAAI/bge-small-zh-v1.5')
    
    def get_text_embedding(self, text: str) -> np.ndarray:
        return self.model.encode(text) 